#!/bin/bash

# Nuclei模块测试脚本
echo "=== 开始Nuclei扫描引擎功能测试 ==="
echo ""

# 检查是否有参数（测试目标URL）
TARGET_URL="http://httpbin.org"  # 默认测试目标
if [ $# -gt 0 ]; then
    TARGET_URL="$1"
fi

echo "📋 测试目标: $TARGET_URL"
echo "📝 POC目录: ./rules/pocs"
echo ""

# 显示可用的POC文件
echo "🔍 可用的POC文件:"
ls -la ./rules/pocs/*.yaml 2>/dev/null || echo "  - 未找到POC文件"
echo ""

# 运行测试
echo "🚀 启动Nuclei功能验证..."
echo ""

# 确保在正确的目录
export RUST_LOG=info

# 执行测试
# 使用我们已验证能正常工作的test_nuclei二进制文件
cd nuclei && cargo run --bin test_nuclei && cd ..

# 如果用户指定了目标URL，可以添加自定义扫描选项
echo ""
echo "🎉 测试完成!"
echo ""
echo "📌 提示:"
echo "  - 要扫描自定义目标: ./test_nuclei.sh http://your-target.com"
echo "  - 要添加新的POC: 将YAML文件放入 ./rules/pocs 目录"
echo "  - 要查看更多日志: export RUST_LOG=debug 后再运行"
echo ""
echo "=== 测试结束 ==="
